#include <iostream>
#include <vector>

using namespace std;
// 63. 不同路径 II
//一个机器人位于一个 m x n 网格的左上角 （起始点在下图中标记为 “Start” ）。
//机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角（在下图中标记为 “Finish”）。
//现在考虑网格中有障碍物。那么从左上角到右下角将会有多少条不同的路径？
//网格中的障碍物和空位置分别用 1 和 0 来表示。

class Solution {
public:
    int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {
        int m = obstacleGrid.size();
        if(m == 0) return 0;

        int n = obstacleGrid[0].size();
        if(n == 0 || obstacleGrid[0][0])    // 起点即障碍，游戏结束
            return 0;
        // 定义状态 dp(i, j) = dp(i+1, j) + dp(i, j+1)
        vector<vector<long long>> dp(m, vector<long long>(n, 0));
        for(int i = m - 1; i >= 0; i--)
            for(int j = n - 1; j >= 0; j--) {
                if(obstacleGrid[i][j] == 1) // 障碍物
                    dp[i][j] = 0;
                else if(i == m - 1 && j == n - 1)   // 终点
                    dp[i][j] = 0;
                else if(i == m - 1 || j == n - 1)   // 下边缘 & 右边缘
                    dp[i][j] = 1;
                else
                    dp[i][j] = dp[i+1][j] + dp[i][j+1]; // 递推
            }

        return dp[0][0];
    }
};

int main() {
    // 初始化网格
    vector<vector<int>> obstacleGrid{
            {0,0,0},
            {0,1,0},
            {0,0,0}
    };

    cout << "obstacleGrid:" << endl;
    for (const auto& row : obstacleGrid) {
        for (const auto& element : row) {
            cout << element << "\t";
        }
        cout << endl;
    }

    cout << "answer: " << Solution().uniquePathsWithObstacles(obstacleGrid) << endl;

    vector<vector<int>> obstacleGrid2{
            {0,1},
            {0,0}
    };
    cout << "answer: " << Solution().uniquePathsWithObstacles(obstacleGrid2) << endl;
    return 0;
}
